package config

import (
	_ "github.com/spf13/cast"
	"github.com/spf13/viper"
)

func init() {
	LoadConfig()
}

// LoadConfig 加载配置文件
func LoadConfig() {
	// 设置配置文件名
	//configName := fmt.Sprintf("%s-%s", "config", ServerStartupFlags.Environment)
	viper.SetConfigName("config")
	// 设置配置文件路径
	viper.AddConfigPath("config")
	// 测试时使用路径
	viper.AddConfigPath("config")
	// 解析配置
	viper.ReadInConfig()
}

// GetDBConfig 获取db配置
func GetDBConfig() map[string]interface{} {
	return viper.GetStringMap("db")
}

func GetDBConfigConnect() map[string]interface{} {
	return viper.GetStringMap("dbConfig")
}
